op {
  graph_op_name: "Diag"
  endpoint {
    name: "linalg.tensor_diag"
  }
  endpoint {
    name: "diag"
    deprecation_version: 2
  }
}
